package com.framework.common;

import com.alibaba.fastjson.JSON;
import com.framework.common.exception.CustomException;
import com.framework.common.utils.StringUtils;
import com.framework.utils.IpUtil;
import lombok.extern.slf4j.Slf4j;

import javax.servlet.http.HttpServletRequest;
import java.util.HashMap;
import java.util.Set;

@Slf4j
public class AjaxResult extends HashMap<String, Object> {
    private static final long serialVersionUID = 1L;

    public static final String CODE_TAG = "code";

    public static final String MSG_TAG = "msg";

    public static final String DATA_TAG = "data";

    /**
     * 状态类型
     */
    public enum Type {

        /**
         * 成功
         */
        SUCCESS(200),
        /**
         * 警告
         */
        WARN(201),
        /**
         * 错误
         */
        ERROR(500);
        private final int value;

        Type(int value) {
            this.value = value;
        }

        public int value() {
            return this.value;
        }
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象，使其表示一个空消息。
     */
    public AjaxResult() {
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     *
     * @param type 状态类型
     * @param msg  返回内容
     */
    public AjaxResult(Type type, String msg) {
        super.put(CODE_TAG, type.value);
        super.put(MSG_TAG, msg);
    }

    /**
     * 初始化一个新创建的 AjaxResult 对象
     *
     * @param type 状态类型
     * @param msg  返回内容
     * @param data 数据对象
     */
    public AjaxResult(Type type, String msg, Object data) {
        super.put(CODE_TAG, type.value);
        super.put(MSG_TAG, msg);
        if (StringUtils.isNotNull(data)) {
            super.put(DATA_TAG, data);
        }
    }

    /**
     * 返回成功消息
     *
     * @return 成功消息
     */
    public static AjaxResult success() {
        return AjaxResult.success("");
    }

    /**
     * 返回成功数据
     *
     * @return 成功消息
     */
    public static AjaxResult success(Object data) {
        return AjaxResult.success("", data);
    }

    /**
     * 返回成功消息
     *
     * @param msg 返回内容
     * @return 成功消息
     */
    public static AjaxResult success(String msg) {
        return AjaxResult.success(msg, null);
    }

    /**
     * 返回成功消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 成功消息
     */
    public static AjaxResult success(String msg, Object data) {
        return success(msg, data, null);
    }
    public static AjaxResult success(String msg, Object data, HttpServletRequest httpServletRequest) {
        try {
            String str = JSON.toJSONString(data);
            if(str.length() > 1024) {
                if(httpServletRequest != null) {
                    Set<String> ips = IpUtil.getIpAddr(httpServletRequest);
                    log.info("result success(ip,msg,obj.len,obj):{},{},{},{}",JSON.toJSONString(ips), msg,str.length(),str.substring(0, 1024));
                } else {
                    log.info("result success(msg,obj.len,obj):{},{},{}",msg,str.length(),str.substring(0, 1024));
                }
            } else {
                if(httpServletRequest != null) {
                    Set<String> ips = IpUtil.getIpAddr(httpServletRequest);
                    log.info("result success(ip,msg,obj.len,obj):{}, {},{},{}",JSON.toJSONString(ips), str.length(),msg,str);
                } else {
                    log.info("result success(msg,obj.len,obj):{},{},{}",msg,str.length(),str);
                }
            }
        }catch (Exception ex) {
            ex.printStackTrace();
        }
        return new AjaxResult(Type.SUCCESS, msg, data);
    }

    /**
     * 返回警告消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static AjaxResult warn(String msg) {
        return AjaxResult.warn(msg, null);
    }

    /**
     * 返回警告消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static AjaxResult warn(String msg, Object data) {
        try {
            String str = JSON.toJSONString(data);
            if(str.length() > 1024) {
                log.info("result warn(msg,obj):"+ msg+str.substring(0, 1024));
            } else {
                log.info("result warn(msg,obj):"+ msg+str);
            }
        }catch (Exception ex) {
            ex.printStackTrace();
        }
        return new AjaxResult(Type.WARN, msg, data);
    }

    /**
     * 返回错误消息
     *
     * @return
     */
    public static AjaxResult error() {
        return AjaxResult.error("操作失败");
    }

    /**
     * 返回错误消息
     *
     * @param msg 返回内容
     * @return 警告消息
     */
    public static AjaxResult error(String msg) {
        return AjaxResult.error(msg, null);
    }

    /**
     * 返回错误消息
     *
     * @param msg  返回内容
     * @param data 数据对象
     * @return 警告消息
     */
    public static AjaxResult error(String msg, Object data) {
        try {
            String str = JSON.toJSONString(data);
            if(str.length() > 1024) {
                log.info("result error(msg,obj):"+ msg+str.substring(0, 1024));
            } else {
                log.info("result error(msg,obj):"+ msg+str);
            }
        }catch (Exception ex) {
            ex.printStackTrace();
        }
        return new AjaxResult(Type.ERROR, msg, data);
    }

    /**
     * 返回异常信息构造函数
     *
     * @param resultCode 自定义异常
     */
    public AjaxResult(ResultCode resultCode){
        super.put(CODE_TAG, resultCode.getCode());
        super.put(MSG_TAG, resultCode.getMsg());
    }


    /**
     * 返回异常信息构造函数(自定义msg)
     *
     * @param resultCode 自定义异常
     */
    public AjaxResult(ResultCode resultCode, String msg){
        super.put(CODE_TAG, resultCode.getCode());
        super.put(MSG_TAG, msg);
    }
}


